package com.hlf.modules.base.entity;


import com.hlf.common.exception.BusinessException;
import com.hlf.common.utils.IdWorkerFactory;

/**
 * model工具类
 *
 */
public class ModelUtils {

    /**
     * 检查id，如果id为空则生成
     *
     * @param record
     *            BaseModel
     * @return 参数对象
     * @author shuliangxing
     * @date 2017年3月24日 下午4:27:11
     */
    public static BaseModel checkOrAddId(BaseModel record) {
        if (record.getId() == null) {
            record.setId(IdWorkerFactory.nextId());
        }
        return record;
    }

    /**
     * 检查id，如果id为空则抛出 {@link BusinessException}
     *
     * @param record
     *            BaseModel
     * @return 参数对象
     * @author shuliangxing
     * @date 2017年3月24日 下午4:27:11
     */
    public static BaseModel checkIdNull(BaseModel record) {
        if (record.getId() == null) {
            throw new BusinessException("主键不能为空");
        }
        return record;
    }
}